package mydb

import (
	"fmt"
	"gitee.com/DonHz/cloud-file-storage/model"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var DB *gorm.DB

func init() {
	dsn := "root:123456@tcp(127.0.0.1:3306)/file-store?charset=utf8mb4&parseTime=True&loc=Local"
	var err error
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		panic(err)
	}
	sqlDB, err := DB.DB() // 连接池
	if err != nil {
		panic(err)
	}
	sqlDB.SetMaxIdleConns(10)  // 最大空闲连接数
	sqlDB.SetMaxOpenConns(100) // 最大连接数
	err = sqlDB.Ping()
	if err != nil {
		fmt.Println("failed to connect to mysql ")
		panic(err)
	}
	DB.AutoMigrate(&model.TableFile{}, &model.User{}, &model.UserToken{}, &model.UserFile{})
}
